package com.storm.queue

import org.springframework.data.redis.core.RedisTemplate
import org.springframework.util.StringUtils

class RedisLock(private val redisTemplate: RedisTemplate<String, Any>) {
    /**
     * 加锁
     *
     * @param key   锁类型
     * @param value 当前时间+超时时间 也就是时间戳
     * @return
     */
    fun lock(key: String, value: String): Boolean {
        if (redisTemplate.opsForValue().setIfAbsent(key, value) == true) { //对应setnx命令
            //可以成功设置,也就是key不存在
            return true
        }
        //判断锁超时 - 防止原来的操作异常，没有运行解锁操作  防止死锁
        val currentValue = (redisTemplate.opsForValue().get(key) ?: return false) as String
        //如果锁过期
        if (!StringUtils.isEmpty(currentValue) && currentValue.toLong() < System.currentTimeMillis()) { //currentValue不为空且小于当前时间
            //获取上一个锁的时间value
            val oldValue = redisTemplate.opsForValue().getAndSet(key, value) as String //对应getset，如果key存在
            //假设两个线程同时进来这里，因为key被占用了，而且锁过期了。获取的值currentValue=A(get取的旧的值肯定是一样的),两个线程的value都是B,key都是K.锁时间已经过期了。
            //而这里面的getAndSet一次只会一个执行，也就是一个执行之后，上一个的value已经变成了B。只有一个线程获取的上一个值会是A，另一个线程拿到的值是B。
            if (!StringUtils.isEmpty(oldValue) && oldValue == currentValue) {
                //oldValue不为空且oldValue等于currentValue，也就是校验是不是上个对应的商品时间戳，也是防止并发
                return true
            }
        }
        return false
    }


    /**
     * 解锁
     * @param key
     * @param value
     */
    fun unlock(key: String, value: String) {
        try {
            val currentValue = redisTemplate.opsForValue().get(key) as String
            if (!StringUtils.isEmpty(currentValue) && currentValue == value) {
                redisTemplate.opsForValue().operations.delete(key) //删除key
            }
        } catch (e: Exception) {

        }
    }

    fun getLockValue(): String {
        return (System.currentTimeMillis() + 10000).toString()
    }
}